#coding:utf-8
import json
import os
from locust import HttpUser, TaskSet, task, between

class WebsiteTasks(TaskSet):
    def on_start(self):
        self.client.post("/login",{
            "username":'test',
            "password":"888888"
        })
    @task(2)
    def index(self):
        self.client.get("/")
    @task(1)
    def about(self):
        self.client.get("/about")
class WebsiteUser(HttpUser):
    task_create=WebsiteTasks
    host = 'https://debugtalk.com/'  # 目标服务地址
    min_wait = 1000  # 单位为ms  最小等待时间  最新版本 已弃用 （当前版本:0.14.5）
    max_wait = 5000  # 单位为ms  最大等待时间  最新版本 已弃用 （当前版本:0.14.5）
    # between(min_wait, max_wait)
    wait_time = between(2, 5)  # 单位为s   等待时间  任务执行间隔时间


# 以下 便于当前脚本 本地调试
# 启动 当前脚本
if __name__ == "__main__":
    cmd = 'locust -f baidu.py'
    os.system(cmd)